package org.basic.algorithm.medium;

import org.basic.common.entity.ListNode;

/**
 * 给你两个 非空 的链表，表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的，并且每个节点只能存储 一位 数字。
 * 你可以假设除了数字 0 之外，这两个数都不会以 0 开头。
 * 请你将两个数相加，并以相同形式返回一个表示和的链表。
 * <p>
 * 示例 1：
 * 输入：l1 = [2,4,3], l2 = [5,6,4]
 * 输出：[7,0,8]
 * 解释：342 + 465 = 807.
 * <p>
 * 示例 2：
 * 输入：l1 = [0], l2 = [0]
 * 输出：[0]
 * <p>
 * 示例 3：
 * 输入：l1 = [9,9,9,9,9,9,9], l2 = [9,9,9,9]
 * 输出：[8,9,9,9,0,0,0,1]
 * <p>
 * 提示：
 * 每个链表中的节点数在范围 [1, 100] 内
 * 0 <= Node.val <= 9
 * 题目数据保证列表表示的数字不含前导零
 *
 * @author LiKun
 * @date 2021/10/8 18:52
 */
public class AddTwoNumbers {

    public static void main(String[] args) {
        ListNode listNode1 = new ListNode(9, new ListNode(9, new ListNode(3)));
        ListNode listNode2 = new ListNode(9, null);

        System.out.println(AddTwoNumbers.addTwoNumbers(listNode1, listNode2));
    }

    public static ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        return AddTwoNumbers.addTwoNumbersSelf(l1, l2);
    }

    /**
     * 执行速度中等，内存占用中等
     * 时间：2 ms；内存：38.5-38.9 MB
     *
     * @param l1 链表
     * @param l2 链表
     *
     * @return 合并后链表
     */
    public static ListNode addTwoNumbersSelf(ListNode l1, ListNode l2) {
        int sum;
        boolean isCarry = false;
        ListNode result = null, nextNode = null;

        while (l1 != null && l2 != null) {
            sum = isCarry ? l1.val + l2.val + 1 : l1.val + l2.val;

            if (sum > 9) {
                sum = sum - 10;
                isCarry = true;
            } else {
                isCarry = false;
            }

            if (result == null) {
                result = new ListNode(sum);
                nextNode = result;
            } else {
                nextNode = growthLength(nextNode, sum);
            }

            l2 = l2.next;
            l1 = l1.next;
        }

        if (l1 != null) {
            calListNode(nextNode, l1, isCarry);
        } else {
            calListNode(nextNode, l2, isCarry);
        }

        return result;
    }

    private static void calListNode(ListNode lastNode, ListNode list, boolean isCarry) {
        while (list != null) {
            if (isCarry) {
                isCarry = list.val == 9;

                lastNode = growthLength(lastNode, isCarry ? 0 : list.val + 1);
                list = list.next;
            } else {
                lastNode.next = list;
                break;
            }
        }

        if (isCarry) {
            lastNode.next = new ListNode(1);
        }
    }

    private static ListNode growthLength(ListNode lastNode, int value) {
        lastNode.next = new ListNode(value);
        lastNode = lastNode.next;
        return lastNode;
    }

}